package com.fly.feishu.web.helper;

import com.fly.feishu.api.dto.WebAccessToken;

/**
 * 飞书登陆用户身份信息
 */
public class FeishuUserContext {
    private static ThreadLocal<WebAccessToken> webAccessTokenThreadLocal = new ThreadLocal<>();

    public static WebAccessToken getCurrentWebToke() {
        if (webAccessTokenThreadLocal != null) {
            return webAccessTokenThreadLocal.get();
        }
        return null;
    }

    public static void setWebAccessTokenThreadLocal(WebAccessToken webAccessToken) {
        if (webAccessTokenThreadLocal == null) {
            webAccessTokenThreadLocal = new ThreadLocal<>();
        }
        webAccessTokenThreadLocal.set(webAccessToken);
    }

    public static void clear() {
        if (webAccessTokenThreadLocal != null && webAccessTokenThreadLocal.get() != null) {
            webAccessTokenThreadLocal.remove();
        }
    }
}
